#pragma once

#include <opencv2\opencv.hpp>

using namespace cv;

class Bayes
{
private:
	CvNormalBayesClassifier bayes;

public:

	void set_trainer(CvNormalBayesClassifier _bayes)
	{
		bayes = _bayes;
	}

	CvNormalBayesClassifier get_trainer()
	{
		return bayes;
	}

	void train(const Mat& data, const Mat& label, const CvNormalBayesClassifier& params)
	{
		bayes.train( data, label);
	}

	float predict(const Mat& sample)
	{
		return bayes.predict(sample);
	}
};